package com.chilly.linkedlist;

import java.util.List;

/**
 * 21. 合并两个有序链表
 * 将两个升序链表合并为一个新的 升序链表并返回。新链表是通过拼接给定的两个链表的所有节点组成的。
 * Created by Chilly Cui on 2020/11/27.
 */
public class MergeTwoSortedLists {

    public static void main(String[] args) {
        ListNode listNode = new ListNode(1);
        ListNode listNode2 = new ListNode(2);
        ListNode listNode3 = new ListNode(4);
        listNode.next = listNode2;
        listNode2.next = listNode3;
        listNode3.next = null;


        ListNode listNode11 = new ListNode(1);
        ListNode listNode22 = new ListNode(3);
        ListNode listNode33 = new ListNode(4);
        listNode11.next = listNode22;
        listNode22.next = listNode33;
        listNode33.next = null;

        ListNode mergeTwoLists = new Solution().mergeTwoLists(listNode, listNode11);
        System.out.println(mergeTwoLists);
    }


    static class Solution {

        // 遍历
        // 当 l1 和 l2 都不是空链表时，判断 l1 和 l2 哪一个链表的头节点的值更小，将较小值的节点添加到结果里，
        // 当一个节点被添加到结果里之后，将对应链表中的节点向后移一位。
        public ListNode mergeTwoLists(ListNode l1, ListNode l2) {

            //哨兵节点 prehead ，这可以在最后让我们比较容易地返回合并后的链表
            ListNode prehead = new ListNode(-1);

            //prev 指针，调整它的 next 指针
            ListNode prev = prehead;

            //重复以下过程，直到 l1 或者 l2 指向了 null ：
            // 如果 l1 当前节点的值小于等于 l2 ，我们就把 l1 当前的节点接在 prev 节点的后面同时将 l1 指针往后移一位。
            // 否则，我们对 l2 做同样的操作。不管我们将哪一个元素接在了后面，我们都需要把 prev 向后移一位。
            while (l1 != null && l2 != null) {
                if (l1.val <= l2.val) {
                    prev.next = l1;
                    l1 = l1.next;
                } else {
                    prev.next = l2;
                    l2 = l2.next;
                }

                prev = prev.next;
            }

            // 合并后 l1 和 l2 最多只有一个还未被合并完，我们直接将链表末尾指向未合并完的链表即可
            prev.next = l1 == null ? l2 : l1;

            return prehead.next;
        }

        /*
        //递归
        public ListNode mergeTwoLists(ListNode l1, ListNode l2) {
            if (l1 == null) {
                return l2;
            } else if (l2 == null) {
                return l1;
            } else {
                if (l1.val < l2.val) {
                    l1.next = mergeTwoLists(l1.next, l2);
                    return l1;
                } else {
                    l2.next = mergeTwoLists(l2.next, l1);
                    return l2;
                }
            }
        }*/
    }

    static class ListNode {
        int val;
        ListNode next;

        public ListNode() {
        }

        public ListNode(int val) {
            this.val = val;
        }

        public ListNode(int val, ListNode next) {
            this.val = val;
            this.next = next;
        }
    }
}
